from sqlalchemy.orm import Session
from . import models, schemas

def get_song_by_source_url(db: Session, source_url: str):
    """
    根据 source_url 从数据库中检索歌曲。
    """
    return db.query(models.Song).filter(models.Song.source_url == source_url).first()

def create_song(db: Session, song: schemas.SongCreate):
    """
    在数据库中创建一条新的歌曲记录。
    """
    db_song = models.Song(
        title=song.title,
        artist=song.artist,
        source_url=song.source_url,
        status=models.SongStatus.PENDING
    )
    db.add(db_song)
    db.commit()
    db.refresh(db_song)
    return db_song 